Continuous Subarray Sum

Leetcode #532 | Medium | Префикс | Хэш-таблицы | Математика

Идея

Любимая идея префиксных сумм. Сумам любого подмассива это разность префиксных сумм - значит они должны давать одинаковый остаток при делении на k. Считаем очередную преф сумму, если ее нет в мапе кладем ее и индекс (получится что запишется минимальный индекс, это нужно чтобы максимизировать длину чтобы условие точно выполнилось), а если такой остаток уже есть, то смотрим на длину подмассива и если >=2 под условие, то возвращаем true.

Big-O

• Время O(N)
• Память O(K)

Код

class Solution {
    public boolean checkSubarraySum(int[] nums, int k) {
        Map<Integer, Integer> map = new HashMap<>();
        map.put(0, -1);
        int pref = 0;
        for (int i = 0; i < nums.length; i++) {
            pref += nums[i];
            int rem = pref % k;
            if (!map.containsKey(rem)) map.put(rem, i);
            else if (i - map.get(rem) >= 2) return true;
        }
        return false;
    }
}